SAE on toy model (showing hidden states & reconstructions)